#
# Ronin Exploits - A Ruby library for Ronin that provides exploitation and
# payload crafting functionality.
#
# Copyright (c) 2007-2013 Hal Brodigan (postmodern.mod3 at gmail.com)
#
# This file is part of Ronin Exploits.
#
# Ronin is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Ronin is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Ronin.  If not, see <http://www.gnu.org/licenses/>
#

require 'dm-core'

module Ronin
  module Model
    #
    # Adds a `default_port` property to a model which receives it's default
    # value from a `DEFAULT_PORT` constant.
    #
    module HasDefaultPort
      def self.included(base)
        base.module_eval do
          # Default port to use
          property :default_port, Integer, default: proc { |resource,port|
            if resource.class.const_defined?('DEFAULT_PORT')
              resource.class.const_get('DEFAULT_PORT')
            end
          }
        end
      end
    end
  end
end
